package leetcode.hash和set;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * 单词规律
 *
 * 给定一种规律 pattern 和一个字符串 str ，判断 str 是否遵循相同的规律。
 *
 * 这里的 遵循 指完全匹配，例如， pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应规律。
 *
 * 示例1:
 *
 * 输入: pattern = "abba", str = "dog cat cat dog"
 * 输出: true
 *
 * 解法：
 * 用 map 做 pattern --> str 的映射关系
 * 再用 set 做数据去重
 */
public class Test290单词规律 {
    public static boolean wordPattern(String pattern, String str) {
        Map<Character, String> map = new HashMap<>();
        char[] patternChar = pattern.toCharArray();
        String[] strings = str.split(" ");
        if (patternChar.length != strings.length) {
            return false;
        }
        int i = 0;
        Set<String> set = new HashSet<>();
        while (i < patternChar.length) {
            if (!map.containsKey(patternChar[i])) {
                if (!set.contains(strings[i])) {
                    map.put(patternChar[i], strings[i]);
                    set.add(strings[i]);
                } else {
                    return false;
                }
            } else {
                String val = map.get(patternChar[i]);
                if (!val.equals(strings[i])) {
                    return false;
                }
            }
            i++;
        }
        return true;
    }

    public static void main(String[] args) {
        String pattern = "abba";
//        String str = "dog cat cat fish";
        String str = "dog cat cat dog";
//        String str = "dog dog dog dog";
        System.out.println(wordPattern(pattern, str));
    }
}
